# llm_registry.py
llm_registry = {}

def register_llm(name):
    def decorator(cls):
        llm_registry[name] = cls
        return cls
    return decorator
